package datastructure.sort;

import java.util.Arrays;

/**
 * 基数排序
 *
 * @author jack.wu
 * @version 1.0
 * @date 2020-03-26
 */
public class RadixSort {
    public static void main(String[] args) {

        int[] array = new int[]{5, 1, 9, 20, 4, 10};
        long startTime = System.currentTimeMillis();
        radixSort(array);
        long endTime = System.currentTimeMillis();
        System.out.println(Arrays.toString(array));
        System.out.println("耗时：" + (endTime - startTime));
    }

    /**
     * 基数排序
     *
     * @param array
     */
    private static void radixSort(int[] array) {
        // 1、得到数组中最大数的位数 假设第一个是最大数
        int max = array[0];
        for (int i = 1; i < array.length; i++) {
            if (max < array[i]) {
                max = array[i];
            }
        }
        int maxLength = (max + "").length();

        // 2、定义一个二维数组，每个桶就是一个一维数组 经典空间换时间
        int[][] bucket = new int[10][array.length];
        // 记录每个桶中实际放了多少个数据 比如bucketElementCount[0] = bucket[0] 桶放入数据的个数
        int[] bucketElementCount = new int[10];
        for (int i = 0, n = 1; i < maxLength; i++, n *= 10) {
            // 针对每个元素的对应位数进行处理
            for (int j = 0; j < array.length; j++) {
                // 取出每个元素的个位的值
                int digitOfElement = array[j] / n % 10;
                // 放入对应桶中
                bucket[digitOfElement][bucketElementCount[digitOfElement]] = array[j];
                bucketElementCount[digitOfElement]++;
            }
            // 按照桶中的顺序
            int index = 0;
            for (int k = 0; k < bucketElementCount.length; k++) {
                // 如果桶中有数据才放入
                if (bucketElementCount[k] != 0) {
                    // 循环该桶，放入
                    for (int l = 0; l < bucketElementCount[k]; l++) {
                        // 取出元素放入
                        array[index++] = bucket[k][l];
                    }
                }
                // 每轮处理完之后需要清理
                bucketElementCount[k] = 0;
            }
            System.out.println("第" + (i + 1) + "伦，对应的排序处理 ： " + Arrays.toString(array));
        }
    }
}
